package com.freight.base.utils;

import com.freight.base.dto.AreaDto;

public class AreaDtoUtil<T extends AreaDto> {
    public static <T extends AreaDto> T getAreaDtoSeatch(T t) {
        if (t.getBeginAreaCode() != null) {
            if (t.getBeginAreaCode().endsWith("0000")) {
                t.setBeginAreaCode(t.getBeginAreaCode().substring(0, 2));
                // 市
            } else if (t.getBeginAreaCode().endsWith("00")) {
                t.setBeginAreaCode(t.getBeginAreaCode().substring(0, 4));
                // 区县
            }
        }
        if (t.getEndAreaCode() != null) {
            // 省
            if (t.getEndAreaCode().endsWith("0000")) {
                t.setEndAreaCode(t.getEndAreaCode().substring(0, 2));
                // 市
            } else if (t.getEndAreaCode().endsWith("00")) {
                t.setEndAreaCode(t.getEndAreaCode().substring(0, 4));
                // 区县
            }
        }
        return t;
    }
}

